@echo off
echo 正在启动音乐生成服务...
echo 设置环境变量避免共享内存冲突...

:: 设置环境变量避免Windows共享内存冲突
set HF_HUB_DISABLE_TELEMETRY=1
set TOKENIZERS_PARALLELISM=false
set OMP_NUM_THREADS=1
set CUDA_VISIBLE_DEVICES=0
set TRANSFORMERS_CACHE=C:\transformers_cache
set HF_HOME=C:\hf_home

:: 清理可能的残留缓存
if exist "C:\transformers_cache" (
    echo 清理transformers缓存...
    rmdir /s /q "C:\transformers_cache"
)

:: 使用单线程模式运行，避免多线程问题
echo 正在启动Flask服务...
echo 访问地址: http://localhost:5000
echo API文档: http://localhost:5000/api/music-generation/params

python -m flask run --host=0.0.0.0 --port=5000 --no-reload --no-debug

echo 服务启动完成！
pause